package com.cyzy.utils;

//import org.cyzy.config.CodeProperties;

import org.springframework.stereotype.Component;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.util.Base64;
import java.util.Random;

@Component
public class CodeUtils {
    /*
     * 随机字符字典
     */
    private  final char[] CHARS = { '2', '3', '4', '5', '6', '7', '8', '9', 'A', 'B', 'C', 'D', 'E', 'F', 'G',
            'H', 'J', 'K', 'L', 'M', 'N', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z' };

//    private CodeProperties codeProperties;
//    public CodeUtils(CodeProperties codeProperties){
//        this.codeProperties = codeProperties;
//    }


    /*
     * 随机数
     */
    private Random random = new Random();

    /*
     * 获取4位随机数
     */
    public String getRandomString() {
        StringBuffer buffer = new StringBuffer();
        for (int i = 0; i < 4; i++) {
            buffer.append(CHARS[random.nextInt(CHARS.length)]);
        }
        return buffer.toString();
    }

    /*
     * 获取随机数颜色
     */
    private Color getRandomColor() {
        return new Color(random.nextInt(255), random.nextInt(255), random.nextInt(255));
    }

    /*
     * 返回某颜色的反色
     */
    private Color getReverseColor(Color c) {
        return new Color(255 - c.getRed(), 255 - c.getGreen(), 255-c.getBlue());
    }

    public String getImageOfBase64(String randomString){
//        System.out.printf(properties.getWidth()+","+properties.getHeight());
        int width = 100;
        int height = 30;

//        int width = codeProperties.getWidth();
//        int height = codeProperties.getHeight();

        Color color = getRandomColor();
        Color reverse = getReverseColor(color);

        BufferedImage bi = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
        Graphics2D g = bi.createGraphics();
        g.setFont(new Font(Font.SANS_SERIF, Font.BOLD, 16));
        g.setColor(color);
        g.fillRect(0, 0, width, height);
        g.setColor(reverse);
        g.drawString(randomString, 18, 20);
        for (int i = 0, n = random.nextInt(100); i < n; i++) {
            g.drawRect(random.nextInt(width), random.nextInt(height), 1, 1);
        }

        try {
            ByteArrayOutputStream baos = new ByteArrayOutputStream();
            ImageIO.write(bi,"jpg",baos);
            bi.flush();
            byte[] buffer = baos.toByteArray();
            bi = null;
            return Base64.getEncoder().encodeToString(buffer);

        } catch (IOException e) {
            e.printStackTrace();
        }
        return null;
    }

    public void writeCheckCode(String randomString, OutputStream out) {
//        String randomString = getRandomString();
//        System.out.printf(properties.getWidth()+","+properties.getHeight());
        int width = 80;
        int height = 30;

//        int width = codeProperties.getWidth();
//        int height = codeProperties.getHeight();

        Color color = getRandomColor();
        Color reverse = getReverseColor(color);

        BufferedImage bi = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
        Graphics2D g = bi.createGraphics();
        g.setFont(new Font(Font.SANS_SERIF, Font.BOLD, 16));
        g.setColor(color);
        g.fillRect(0, 0, width, height);
        g.setColor(reverse);
        g.drawString(randomString, 18, 20);
        for (int i = 0, n = random.nextInt(100); i < n; i++) {
            g.drawRect(random.nextInt(width), random.nextInt(height), 1, 1);
        }




        // 转成JPEG格式
//        ServletOutputStream out = response.getOutputStream();
//		JPEGImageEncoder encoder = JPEGCodec.createJPEGEncoder(out);
//		encoder.encode(bi);
        try {
            ImageIO.write(bi,"jpg",out);
            bi.flush();
            bi = null;
            out.flush();
        } catch (IOException e) {
            throw new RuntimeException(e);
        }

    }
}
